import tflite_runtime.interpreter as tflite

class TFLiteInterpreter:
    def __init__(self, model_path):
        self.interpreter = tflite.Interpreter(model_path=model_path)
        self.input_details = self.interpreter.get_input_details()
        self.output_details = self.interpreter.get_output_details()

    def set_input(self, data):
        self.interpreter.set_tensor(self.input_details[0]['index'], data)

    def invoke(self):
        self.interpreter.invoke()

    def get_output(self):
        return self.interpreter.get_tensor(self.output_details[0]['index'])